package com.haoyu.number.secrch;

import java.util.ArrayList;

/**
 * @author: 蒿雨
 * @create: 2022-01-10 19:05
 * @description:二分查找
 */
//注意：使用二分查找法的前提是，该数组是有序的
public class BinarySearch {
    public static void main(String[] args) {
        int arr[] = {1, 8, 10, 89, 1000, 1000, 1000, 1234};
        //int resIndex = binarySearch(arr, 0, arr.length - 1, 1234);
        ArrayList<Integer> list = binarySearch2(arr, 0, arr.length - 1, 1000);
        //System.out.println("resIndex=" + resIndex);
        System.out.println("list=" + list);
    }

    //二分查找法

    /**
     * @param arr     数组
     * @param left    左边的索引
     * @param right   右边的索引
     * @param findVal 要查找的值
     * @return
     */
    public static int binarySearch(int[] arr, int left, int right, int findVal) {
        //中位数
        //当left>right时，说明要递归整个数组，但是没有找到
        if (left > right) {
            return -1;
        }

        int mid = (left + right) / 2;

        int midVal = arr[mid];
        //向右递归
        if (findVal > midVal) {
            return binarySearch(arr, mid + 1, right, findVal);
        }//向左递归
        else if (findVal > midVal) {
            return binarySearch(arr, left, right + 1, findVal);
        } else {
            return mid;
        }
    }

    //完成一个课后思考题：
    /*
     *课后思考题：{1，8，10，89，1000，1000，1234}当一个有序数组中，
     * 有多个相同的数值时，如何将所有的数值都查找到，比如这里的1000
     * 思路分析：
     * 1、再找到mid值，不要马上返回，
     * 2、向mid索引值的左边扫描，将所有满足1000，的元素的下标，加入到集合ArrayList
     * 3、向mid索引值的右边扫描，将所有满足1000，的元素的下标，加入到集合ArrayList
     * 4、将所有的ArrayList返回
     */
    public static ArrayList<Integer> binarySearch2(int[] arr, int left, int right, int findVal) {
        //中位数
        //当left>right时，说明要递归整个数组，但是没有找到
        if (left > right) {
            return new ArrayList<Integer>();
        }

        int mid = (left + right) / 2;
        int midVal = arr[mid];
        //向右递归
        if (findVal > midVal) {
            return binarySearch2(arr, mid + 1, right, findVal);
        }//向左递归
        else if (findVal > midVal) {
            return binarySearch2(arr, left, right + 1, findVal);
        } else {
            ArrayList<Integer> resIndexlist = new ArrayList<>();
            //向mid索引值的左边扫描，将所有满足1000，的元素的下标，加入到集合ArrayList
            int temp = mid - 1;
            while (true) {
                //退出
                if (temp < 0 || arr[temp] != findVal) {
                    break;
                }
                //否则，就temp，放入到resIndexList
                resIndexlist.add(temp);
                //temp左移
                temp -= 1;
            }
            resIndexlist.add(mid);

            //向mid索引值的右边扫描，将所有满足1000，的元素的下标，加入到集合ArrayList
            temp = mid + 1;
            while (true) {
                //退出
                if (temp > arr.length - 1 || arr[temp] != findVal) {
                    break;
                }
                //否则，就temp，放入到resIndexList
                resIndexlist.add(temp);
                //temp右移
                temp += 1;
            }

            return resIndexlist;
        }
    }

}
